;;
;; Copyright (c) 2023, Intel Corporation
;;
;; Redistribution and use in source and binary forms, with or without
;; modification, are permitted provided that the following conditions are met:
;;
;;     * Redistributions of source code must retain the above copyright notice,
;;       this list of conditions and the following disclaimer.
;;     * Redistributions in binary form must reproduce the above copyright
;;       notice, this list of conditions and the following disclaimer in the
;;       documentation and/or other materials provided with the distribution.
;;     * Neither the name of Intel Corporation nor the names of its contributors
;;       may be used to endorse or promote products derived from this software
;;       without specific prior written permission.
;;
;; THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
;; AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
;; IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
;; DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE
;; FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
;; DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
;; SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
;; CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
;; OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
;; OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
;;

;; function to wrap cpuid opcode across OS versions
%include "include/os.inc"
%include "include/reg_sizes.inc"

%ifdef LINUX
%define arg1		rdi
%define arg2	        rsi
%define arg3            rdx
%else
%define arg1		rcx
%define arg2	        rdx
%define arg3            r8
%endif

mksection .text

;;
;; Wrapper for CPUID opcode
;;
;; Parameters:
;;    [in] leaf    - CPUID leaf number (EAX)
;;    [in] subleaf - CPUID sub-leaf number (ECX)
;;    [out] out    - registers structure to store results of CPUID into
;;
;; void mbcpuid(const unsigned leaf, const unsigned subleaf, struct cpuid_regs *out)

MKGLOBAL(mbcpuid,function,internal)
mbcpuid:
        push    rbx

        mov     r11, arg3       ;; arg3 will get overwritten with cpuid on sysv
        mov     eax, DWORD(arg1)
        mov     ecx, DWORD(arg2)

        cpuid

        mov     [r11 + 0*4], eax
        mov     [r11 + 1*4], ebx
        mov     [r11 + 2*4], ecx
        mov     [r11 + 3*4], edx

        pop     rbx
        ret

mksection stack-noexec
